package com.example.controller;

import com.example.utils.R;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.lang.reflect.Field;
import java.util.UUID;

@RestController
@RequestMapping("/common")
public class CommonController {
    @Value("${reggie.image}")
    private String path;
    /**
     * 文件上传
     * @param file 上传的文件
     * @return 响应给页面的数据方便添加菜品时使用
     */
    @PostMapping("/upload")
    public R upload(MultipartFile file)  {
        //获取到文件的原名称
        String originalFilename = file.getOriginalFilename();
        //获取文件的后缀名（文件类型）
        String end = originalFilename.substring(originalFilename.lastIndexOf("."));
        //由于可能会出现重名问题所以要用UUID生成随机的名字(文件名)
        String imagePath = UUID.randomUUID().toString()+end;
        //我们将文件存储到文件夹中所以常见一个File判断文件是否以被创建
        File file1 = new File(path);
        if (!file1.exists()){
            file1.mkdirs();
        }
        try {
            //前端上传的file文件会存储到临时文件中，要将其替换成指定的文件并进行存储
            file.transferTo(new File(path+imagePath));
        } catch (IOException e) {
            e.printStackTrace();
        }
        return new R(1,imagePath,"");
    }

    /**
     * 文件下载（回显图片）
     * @param name 图片名称
     * @param response 响应图片给浏览器
     */
    @GetMapping("/download")
    public void download(String name, HttpServletResponse response){
        //使用输入流，读取文件内容
        FileInputStream fileInputStream = null;
        ServletOutputStream outputStream = null;
        //设置响应文件类型
        response.setContentType("image/jpeg");
        try {
            fileInputStream = new FileInputStream(path+name);
            outputStream = response.getOutputStream();
            //定义读取行数
            int len = 0;
            //定义每次读取的字节数组
            byte[] bytes = new byte[1024];
            //读取数据
            while ((len=fileInputStream.read(bytes))!=-1){
                //写入数据（写回浏览器显示文件）
                outputStream.write(bytes,0,len);
                //刷新数据
                outputStream.flush();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }finally {
            try {
                fileInputStream.close();
                outputStream.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}
